namespace Maple.Branch.Identity.DomainTests
{
    using System;
    using System.Linq;
    using EfCoreConfigurations;
    using Entities;
    using TestBases;
    using Volo.Abp.Domain.Entities;
    using Volo.Abp.Modularity;

    public abstract class IdentityExtendedTestBase<TStartupModule> : IdentityTestBase<TStartupModule>
        where TStartupModule : IAbpModule
    {
        protected virtual IdentityUser GetUser(string userName)
        {
            var user = UsingDbContext(context => context.Users.FirstOrDefault(u => u.UserName == userName));
            if (user == null)
            {
                throw new EntityNotFoundException();
            }

            return user;
        }

        protected virtual IdentityUser FindUser(string userName)
        {
            return UsingDbContext(context => context.Users.FirstOrDefault(u => u.UserName == userName));
        }

        protected virtual void UsingDbContext(Action<IIdentityDbContext> action)
        {
            using var dbContext = GetRequiredService<IIdentityDbContext>();

            action.Invoke(dbContext);
        }

        protected virtual T UsingDbContext<T>(Func<IIdentityDbContext, T> action)
        {
            using var dbContext = GetRequiredService<IIdentityDbContext>();

            return action.Invoke(dbContext);
        }
    }
}
